'''
author:
desc:是对selenium api 的二次封装,将自带的方法进行重新封装
date:2019/11/29 9:06
'''
from selenium import webdriver

from selenium.webdriver.common.action_chains import ActionChains  # 鼠标相关包

from selenium.webdriver.support import expected_conditions as EC  # 获取 元素等待需要导入的包

from selenium.webdriver.support.ui import WebDriverWait  # 获取 元素等待需要导入的包

from selenium.webdriver.common.by import By  # 获取 元素等待需要导入的包

from selenium.webdriver.support.select import Select  # select tag 相关包


class Pyse(object):
	'''

	        Pyse framework for the main class, the original

	    selenium provided by the method of the two packaging,

	    making it easier to use.

	    '''

	def __init__(self, browser='ff'):  # 需要实例化参数,默认firefox

		'''

		Run class initialization method, the default is proper

		to drive the Firefox browser. Of course, you can also

		pass parameter for other browser, Chrome browser for the "Chrome",

		the Internet Explorer browser for "internet explorer" or "ie".
		前提将不同的浏览器驱动配置好环境变量

		'''
		if browser == "firefox" or browser == "ff":
			driver = webdriver.Firefox(executable_path=r"D:\Web\geckodriver")

		elif browser == "chrome":

			option = webdriver.ChromeOptions()  # 这是chrome提供的几个配置文件

			option.add_argument("--start-maximized")  # 增加配置文件，这里指启动时直接全屏

			driver = webdriver.Chrome(chrome_options=option)  # 实例化时直接传进来即可

		elif browser == "internet explorer" or browser == "ie":

			driver = webdriver.Ie()

		elif browser == "opera":

			driver = webdriver.Opera()

		elif browser == "phantomjs":

			driver = webdriver.PhantomJS()

		elif browser == 'edge':

			driver = webdriver.Edge()

		try:

			self.driver = driver

		except Exception:

			raise NameError(

				"Not found %s browser,You can enter 'ie', 'ff', 'opera', 'phantomjs', 'edge' or 'chrome'." % browser)

	def element_wait(self, css, secs=20):

		'''

		Waiting for an element to display.
		Usage:
		driver.element_wait("css=>#el",10)
		sec 为超时时间timeout 单位为s
		'''

		if "=>" not in css:
			raise NameError("Positioning syntax errors, lack of '=>'.")

		by = css.split("=>")[0]

		value = css.split("=>")[1]

		if by == "id":

			WebDriverWait(self.driver, secs).until(EC.presence_of_element_located((By.ID, value)))  # 参数是一个元组

		elif by == "name":

			WebDriverWait(self.driver, secs).until(EC.presence_of_element_located((By.NAME, value)))

		elif by == "class":

			WebDriverWait(self.driver, secs).until(EC.presence_of_element_located((By.CLASS_NAME, value)))

		elif by == "link_text":

			WebDriverWait(self.driver, secs).until(EC.presence_of_element_located((By.LINK_TEXT, value)))

		elif by == "xpath":

			WebDriverWait(self.driver, secs).until(EC.presence_of_element_located((By.XPATH, value)))

		elif by == "css":

			WebDriverWait(self.driver, secs).until(EC.presence_of_element_located((By.CSS_SELECTOR, value)))

		else:

			raise NameError(

				"Please enter the correct targeting elements,'id','name','class','link_text','xpath','css'.")

	def element_location(self, css):
		# 返回的是x y 字典
		location = self.get_element(css).location
		return location

	def get_element(self, css):  # 获取定位元素

		'''

		Judge element positioning way, and returns the element.

		'''

		if "=>" not in css:
			raise NameError("Positioning syntax errors, lack of '=>'.")

		by = css.split("=>")[0]

		value = css.split("=>")[1]

		if by == "id":

			element = self.driver.find_element_by_id(value)

		elif by == "name":

			element = self.driver.find_element_by_name(value)

		elif by == "class":

			element = self.driver.find_element_by_class_name(value)

		elif by == "link_text":

			element = self.driver.find_element_by_link_text(value)

		elif by == "xpath":

			element = self.driver.find_element_by_xpath(value)

		elif by == "css":

			element = self.driver.find_element_by_css_selector(value)  # 通过css方式定位

		else:

			raise NameError(

				"Please enter the correct targeting elements,'id','name','class','link_text','xpath','css'.")

		return element  ###注意这里需要返回元素的值。

	def get_elements(self, css):  # 获取定位元素

		'''

		Judge element positioning way, and returns the element.

		'''

		if "=>" not in css:
			raise NameError("Positioning syntax errors, lack of '=>'.")

		by = css.split("=>")[0]

		value = css.split("=>")[1]

		if by == "id":

			element = self.driver.find_elements_by_id(value)

		elif by == "name":

			element = self.driver.find_elements_by_name(value)

		elif by == "class":

			element = self.driver.find_elements_by_class_name(value)

		elif by == "link_text":

			element = self.driver.find_elements_by_link_text(value)

		elif by == "xpath":

			element = self.driver.find_elements_by_xpath(value)

		elif by == "css":

			element = self.driver.find_elements_by_css_selector(value)  # 通过css方式定位

		else:

			raise NameError(

				"Please enter the correct targeting elements,'id','name','class','link_text','xpath','css'.")

		return element  ###注意这里需要返回元素的值。

	def open(self, url):

		'''
		open url.
		Usage:
		driver.open("https://www.baidu.com")
		'''

		self.driver.get(url)  # 只是一个方法是一个函数。源码是post

	def max_window(self):

		'''
		Set browser window maximized.
		Usage:
		driver.max_window()
		'''

		self.driver.maximize_window()

	def set_window(self, wide, high):

		'''

		Set browser window wide and high.

		Usage:

		driver.set_window(wide,high)

		'''

		self.driver.set_window_size(wide, high)

	def type(self, css, text):

		'''

		Operation input box.



		Usage:

		driver.type("css=>#el","selenium")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		el.send_keys(text)

	def clear(self, css):

		'''

		Clear the contents of the input box.



		Usage:

		driver.clear("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		el.clear()

	def click(self, css):

		'''

		It can click any text / image can be clicked

		Connection, check box, radio buttons, and even drop-down box etc..



		Usage:

		driver.click("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		el.click()

	def right_click(self, css):

		'''

		Right click element.



		Usage:

		driver.right_click("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		ActionChains(self.driver).context_click(el).perform()

	def move_to_element(self, css):

		'''

		Mouse over the element.



		Usage:

		driver.move_to_element("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		ActionChains(self.driver).move_to_element(el).perform()

	def double_click(self, css):

		'''

		Double click element.



		Usage:

		driver.double_click("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		ActionChains(self.driver).double_click(el).perform()

	def drag_and_drop(self, el_css, ta_css):

		'''

		Drags an element a certain distance and then drops it.



		Usage:

		driver.drag_and_drop("css=>#el","css=>#ta")

		'''

		self.element_wait(el_css)

		element = self.get_element(el_css)

		self.element_wait(ta_css)

		target = self.get_element(ta_css)

		ActionChains(driver).drag_and_drop(element, target).perform()

	def click_text(self, text):

		'''

		Click the element by the link text



		Usage:

		driver.click_text("新闻")

		'''

		self.driver.find_element_by_partial_link_text(text).click()

	def close(self):

		'''

		Simulates the user clicking the "close" button in the titlebar of a popup

		window or tab.



		Usage:

		driver.close()

		'''

		self.driver.close()

	def quit(self):

		'''

		Quit the driver and close all the windows.



		Usage:

		driver.quit()

		'''

		self.driver.quit()

	def submit(self, css):

		'''

		Submit the specified form.



		Usage:

		driver.submit("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		el.submit()

	def F5(self):

		'''

		Refresh the current page.



		Usage:

		driver.F5()

		'''

		self.driver.refresh()

	def js(self, script):

		'''

		Execute JavaScript scripts.



		Usage:

		driver.js("window.scrollTo(200,1000);")

		'''

		self.driver.execute_script(script)

	def get_attribute(self, css, attribute):

		'''

		Gets the value of an element attribute.



		Usage:

		driver.get_attribute("css=>#el","type")

		'''

		el = self.get_element(css)

		return el.get_attribute(attribute)

	def get_text(self, css):

		'''

		Get element text information.



		Usage:

		driver.get_text("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		return el.text

	def get_display(self, css):

		'''

		Gets the element to display,The return result is true or false.



		Usage:

		driver.get_display("css=>#el")

		'''

		self.element_wait(css)

		el = self.get_element(css)

		return el.is_displayed()

	def get_title(self):

		'''

		Get window title.



		Usage:

		driver.get_title()

		'''

		return self.driver.title

	def get_url(self):

		'''

		Get the URL address of the current page.



		Usage:

		driver.get_url()

		'''

		return self.driver.current_url

	def get_windows_img(self, file_path):

		'''

		Get the current window screenshot.



		Usage:

		driver.get_windows_img()

		'''

		self.driver.get_screenshot_as_file(file_path)

	def wait(self, secs):

		'''

		Implicitly wait.All elements on the page.



		Usage:

		driver.wait(10)

		'''

		self.driver.implicitly_wait(secs)

	def accept_alert(self):

		'''

		Accept warning box.



		Usage:

		driver.accept_alert()

		'''

		self.driver.switch_to.alert.accept()

	def dismiss_alert(self):

		'''

		Dismisses the alert available.



		Usage:

		driver.dismiss_alert()

		'''

		self.driver.switch_to.alert.dismiss()
    def text_alert(self,text):

		'''

		type the alert available.



		Usage:

		'''

		self.driver.switch_to.alert.send_keys(text)

	def switch_to_frame(self, css):

		'''

		Switch to the specified frame.



		Usage:

		driver.switch_to_frame("css=>#el")

		'''

		self.element_wait(css)

		iframe_el = self.get_element(css)

		self.driver.switch_to.frame(iframe_el)  ###切换 但如果是id name可以直接switch 参入参数即可

	def switch_to_frame_out(self):

		'''

		Returns the current form machine form at the next higher level.

		Corresponding relationship with switch_to_frame () method.



		Usage:

		driver.switch_to_frame_out()

		'''

		self.driver.switch_to.default_content()

	def switch_to_parent_frame(self):

		'''

		Returns the current form machine form at the next higher level.

		Corresponding relationship with switch_to_frame () method.



		Usage:

		driver.switch_to_frame_out()

		'''

		self.driver.switch_to.parent_frame()

	def open_new_window(self, css):

		'''

		Open the new window and switch the handle to the newly opened window.



		Usage:

		driver.open_new_window()

		'''

		original_windows = self.driver.current_window_handle

		el = self.get_element(css)

		el.click()

		all_handles = self.driver.window_handles

		for handle in all_handles:

			if handle != original_windows:
				self.driver.switch_to.window(handle)



			# def _save_png(self, name):

			#     self.get_windows_img(name)

	def wait_and_save_exception(self, css, name):  # 等待并且保存错误截图

		try:

			self.element_wait(css, secs=5)

			return True

		except Exception as e:
			from conf.config import WEBPICTUREPATH  # 文件地址

			self.get_windows_img(WEBPICTUREPATH + name + '.jpg')

			return False

	def wait_and_exception(self, css):

		try:

			self.element_wait(css, secs=10)

			return True

		except Exception as e:

			return False

	def select_by_value(self, css, value):

		self.element_wait(css)

		el = self.get_element(css)

		Select(el).select_by_value(value)


if __name__ == '__main__':  # 便于本文件调试使用，只有在当前页面会执行该代码

	driver = Pyse()
